package Q559;

import java.util.LinkedList;
import java.util.List;
import java.util.Queue;

/**
 * N 叉树的最大深度
 * 给定一个 N 叉树，找到其最大深度。
 * 最大深度是指从根节点到最远叶子节点的最长路径上的节点总数。
 * N 叉树输入按层序遍历序列化表示，每组子节点由空值分隔（请参见示例）。
 */
public class Test {
    public static void main(String[] args) {

    }
}

class Node {
    public int val;
    public List<Node> children;

    public Node() {
    }

    public Node(int _val) {
        val = _val;
    }

    public Node(int _val, List<Node> _children) {
        val = _val;
        children = _children;
    }
};

/**
 * bfs
 */
class Solution {
    public int maxDepth(Node root) {  //层序遍历解法
        if (root == null)
            return 0;
        //只有一层
        if (root.children.size() == 0)
            return 1;
        int depth = 0;
        Queue<Node> queue = new LinkedList<Node>();
        queue.add(root);
        while (!queue.isEmpty()) {
            int count = queue.size();
            depth++;
            while (count > 0) {
                Node node = queue.poll();
                if (node.children.size() != 0) {
                    queue.addAll(node.children);
                }
                count--;
            }
        }
        return depth;
    }
}

/**
 * DFS
 */
class Solution2{
    public int maxDepth(Node root) {  //写深搜
        if (root == null) {
            return 0;
        }
        if(root.children.size() == 0){
            return 1;
        }
        int maxLength = 0;
        //得到孩子
        List<Node> children = root.children;
        //遍历孩子
        for (Node child : children) {
            int i = maxDepth(child);
            maxLength = Math.max(maxLength,i);
        }
        return maxLength+1;
    }
}